"""Add thesis fields

Revision ID: add_thesis_fields
Revises: add_session_token
Create Date: 2024-03-21

"""
from alembic import op
import sqlalchemy as sa

def upgrade():
    op.add_column('thesis', sa.Column('contact_info', sa.String(100), nullable=False))
    op.add_column('thesis', sa.Column('deadline', sa.DateTime, nullable=False))
    op.add_column('thesis', sa.Column('need_environment', sa.Boolean, default=False))

def downgrade():
    op.drop_column('thesis', 'contact_info')
    op.drop_column('thesis', 'deadline')
    op.drop_column('thesis', 'need_environment') 